package com.im.tcp.server;

import com.ecode.imecode.WebSocketMessageDecode;
import com.ecode.imecode.WebSocketMessageEncoder;
import com.im.tcp.feign.RpcServer;
import com.im.tcp.handler.AuthorCheckHandler;
import com.im.tcp.handler.NettyMsgServerHandler;
import com.lld.im.common.config.ImTcpConfig;
import io.netty.bootstrap.ServerBootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioServerSocketChannel;
import io.netty.handler.codec.http.HttpObjectAggregator;
import io.netty.handler.codec.http.HttpServerCodec;
import io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler;
import lombok.extern.slf4j.Slf4j;
import org.redisson.api.RedissonClient;

@Slf4j
public class WebSocketServer {


    private ServerBootstrap b;
    private EventLoopGroup bossGroup;
    private EventLoopGroup workerGroup;

    private RpcServer rpcServer;

    private ImTcpConfig imTcpConfig;

    public WebSocketServer(ImTcpConfig imConfig, RedissonClient redissonClient, RpcServer rpcServer) {
        this.imTcpConfig = imConfig;
        this.rpcServer = rpcServer;
//创建两个线程池 这里使用的是netty的主从线程模型
        // boss负责处理连接 worker负责对 讲chanel分发给handler处理客户端数据接收与回复
        bossGroup = new NioEventLoopGroup(imTcpConfig.getBossGroupPoolSize()); // (1)
        workerGroup = new NioEventLoopGroup(imConfig.getWorkGroupPoolSize());
        //这个类似于一个服务启动器
        b = new ServerBootstrap(); // (2)
        b.group(bossGroup, workerGroup)
                .channel(NioServerSocketChannel.class) // (3)
                .childHandler(new ChannelInitializer<SocketChannel>() { // (4)
                    //一个channel久相当于是一个IO连接 就类似于套接字一样
                    // 当有请求连接进来时 调用方法 新创建一个channel 并交给pipeline执行链 每一个channel都有一个pipeline
                    // 也可以理解为是拦截器 对数据进行拦截处理
                    @Override
                    public void initChannel(SocketChannel ch) throws Exception {
                        // netty提供了对websocket的支持 可以直接使用
                        ChannelPipeline p = ch.pipeline();
                        //处理websocket的http握手请求
                        p.addLast(new HttpServerCodec());
                        //对websocket消息进行聚合
                        p.addLast(new HttpObjectAggregator(65536));
                        //Netty 中的一个关键组件，用于处理 WebSocket 握手和协议升级，以及管理 WebSocket 连接的生命周期
                        p.addLast(new WebSocketServerProtocolHandler("/ws"));
                        //自定义的编解码器
                        p.addLast(new WebSocketMessageDecode());
                        p.addLast(new AuthorCheckHandler(rpcServer));
                        p.addLast(new WebSocketMessageEncoder());
                        p.addLast(new NettyMsgServerHandler(imTcpConfig.getBrokerId(),redissonClient, rpcServer));
                    }
                })
                .option(ChannelOption.SO_BACKLOG, 10240) // 服务端可连接队列大小
                .option(ChannelOption.SO_REUSEADDR, true) // 允许重用本地地址和端口
                .childOption(ChannelOption.TCP_NODELAY, true) // 禁用 Nagle 算法
                .childOption(ChannelOption.SO_KEEPALIVE, true);// 启用 TCP 保持活动消息
    }

    public void start(String host) {
        b.bind(host,imTcpConfig.getWebsocketPort());
        log.info(" Websocket启动:   {}:{}",host,imTcpConfig.getWebsocketPort());
//        log.info("Web Socket Started!");
    }

}
